package com.ybbase.framework.common.util;

import freemarker.template.Configuration;
import freemarker.template.Template;
import org.springframework.util.ResourceUtils;

import java.io.*;
import java.util.Locale;
import java.util.Map;

/**
 * Freemarker 模版引擎类
 *
 * @author shenaiguo
 * @date 2019-12-05
 *
 */
public class FreemarkerUtil {

    public static void main(String[] args) {

    }

    /**
     * 输出到输出到文件
     *
     * @param ftlName  ftl文件名
     * @param root     传入的map
     * @param outFile  输出后的文件全部路径
     * @param filePath 输出前的文件上部路径
     */
    public static void printFile(String ftlName, Map<String, Object> root, String outFile, String filePath, String ftlPath) throws Exception {
        try {
            File file = new File( filePath + outFile);
            //判断有没有父路径，就是判断文件整个路径是否存在
            if (!file.getParentFile().exists()) {
                //不存在就全部创建
                file.getParentFile().mkdirs();
            }
            Writer out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "utf-8"));
            Template template = getTemplate(ftlName, ftlPath);
            //模版输出
            template.process(root, out);
            out.flush();
            out.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 通过文件名加载模版
     *
     * @param ftlName 文件名
     * @param ftlPath 路径名
     */
    public static Template getTemplate(String ftlName, String ftlPath) throws Exception {
        try {
            //通过Freemaker的Configuration读取相应的ftl
            Configuration cfg = new Configuration();
            //设定去哪里读取相应的ftl模板文件
            cfg.setEncoding(Locale.CHINA, "utf-8");
            //在模板文件目录中找到名称为name的文件
         //   cfg.setDirectoryForTemplateLoading(new File(PathUtil.getClassResources()+ftlPath));
            File file = ResourceUtils.getFile("classpath:"+ftlPath);
            cfg.setDirectoryForTemplateLoading(file);
            return cfg.getTemplate(ftlName);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }
}
